my code is ugly: extracting only the files I want from a list of files